# -*- coding: utf-8 -*-
# @Function: 配置文件
# @Description: 存储系统全局配置信息，包括数据库连接参数和水库名称与category_id的映射关系
# @Usage: 被其他模块导入使用，提供统一的配置信息

# 数据库配置
DB_CONFIG = {
    'host': 'localhost',
    'port': 5432,
    'database': 'QA_knowledge',
    'user': 'postgres',
    'password': '12345678',
    'client_encoding': 'UTF8'  # 明确指定UTF-8编码，解决Windows编码问题
}

# 水库名称与category_id的映射关系
RESERVOIR_MAPPING = {
    "白石水库调度规程": "21200510",
    "清河水库调度规程": "20810200",
    "石佛寺水库调度规程": "20600340",
    "大伙房水库调度规程": "21100150",
    "观音阁水库调度规程": "21110150",
    "葠窝水库调度规程": "21110400",
    "柴河水库调度规程": "20800900",
    "汤河水库调度规程": "21113800",
    "闹德海水库调度规程": "20910930",
    "白石": "21200510",
    "清河": "20810200",
    "石佛寺": "20600340",
    "大伙房": "21100150",
    "观音阁": "21110150",
    "葠窝": "21110400",
    "柴河": "20800900",
    "汤河": "21113800",
    "闹德海": "20910930"
}